# Copyright (c) Ansible project
# GNU General Public License v3.0+ (see LICENSES/GPL-3.0-or-later.txt or https://www.gnu.org/licenses/gpl-3.0.txt)
# SPDX-License-Identifier: GPL-3.0-or-later

from __future__ import annotations

import unittest
from unittest.mock import patch

from ansible_collections.community.general.plugins.module_utils.source_control.bitbucket import BitbucketHelper
from ansible_collections.community.general.plugins.modules import bitbucket_access_key
from ansible_collections.community.internal_test_tools.tests.unit.plugins.modules.utils import (
    AnsibleFailJson,
    AnsibleExitJson,
    ModuleTestCase,
    set_module_args,
)


class TestBucketAccessKeyModule(ModuleTestCase):
    def setUp(self):
        super().setUp()
        self.module = bitbucket_access_key

    def test_missing_key_with_present_state(self):
        with self.assertRaises(AnsibleFailJson) as exec_info:
            with set_module_args(
                {
                    "client_id": "ABC",
                    "client_secret": "XXX",
                    "workspace": "name",
                    "repository": "repo",
                    "label": "key name",
                    "state": "present",
                }
            ):
                self.module.main()

        self.assertEqual(exec_info.exception.args[0]["msg"], self.module.error_messages["required_key"])

    @patch.object(bitbucket_access_key, "get_existing_deploy_key", return_value=None)
    def test_create_deploy_key(self, *args):
        with patch.object(self.module, "create_deploy_key") as create_deploy_key_mock:
            with self.assertRaises(AnsibleExitJson) as exec_info:
                with set_module_args(
                    {
                        "user": "ABC",
                        "password": "XXX",
                        "workspace": "name",
                        "repository": "repo",
                        "key": "public_key",
                        "label": "key name",
                        "state": "present",
                    }
                ):
                    self.module.main()

            self.assertEqual(create_deploy_key_mock.call_count, 1)
            self.assertEqual(exec_info.exception.args[0]["changed"], True)

    @patch.object(BitbucketHelper, "fetch_access_token", return_value="token")
    @patch.object(bitbucket_access_key, "get_existing_deploy_key", return_value=None)
    def test_create_deploy_key_check_mode(self, *args):
        with patch.object(self.module, "create_deploy_key") as create_deploy_key_mock:
            with self.assertRaises(AnsibleExitJson) as exec_info:
                with set_module_args(
                    {
                        "client_id": "ABC",
                        "client_secret": "XXX",
                        "workspace": "name",
                        "repository": "repo",
                        "key": "public_key",
                        "label": "key name",
                        "state": "present",
                        "_ansible_check_mode": True,
                    }
                ):
                    self.module.main()

            self.assertEqual(create_deploy_key_mock.call_count, 0)
            self.assertEqual(exec_info.exception.args[0]["changed"], True)

    @patch.object(BitbucketHelper, "fetch_access_token", return_value="token")
    @patch.object(
        bitbucket_access_key,
        "get_existing_deploy_key",
        return_value={
            "id": 123,
            "label": "mykey",
            "created_on": "2019-03-23T10:15:21.517377+00:00",
            "key": "ssh-rsa AAAAB3NzaC1yc2EAAAADA...AdkTg7HGqL3rlaDrEcWfL7Lu6TnhBdq5",
            "type": "deploy_key",
            "comment": "",
            "last_used": None,
            "repository": {
                "links": {
                    "self": {"href": "https://api.bitbucket.org/2.0/repositories/mleu/test"},
                    "html": {"href": "https://bitbucket.org/mleu/test"},
                    "avatar": {"href": "..."},
                },
                "type": "repository",
                "name": "test",
                "full_name": "mleu/test",
                "uuid": "{85d08b4e-571d-44e9-a507-fa476535aa98}",
            },
            "links": {"self": {"href": "https://api.bitbucket.org/2.0/repositories/mleu/test/deploy-keys/123"}},
        },
    )
    def test_update_deploy_key(self, *args):
        with patch.object(self.module, "delete_deploy_key") as delete_deploy_key_mock:
            with patch.object(self.module, "create_deploy_key") as create_deploy_key_mock:
                with self.assertRaises(AnsibleExitJson) as exec_info:
                    with set_module_args(
                        {
                            "client_id": "ABC",
                            "client_secret": "XXX",
                            "workspace": "name",
                            "repository": "repo",
                            "key": "new public key",
                            "label": "mykey",
                            "state": "present",
                        }
                    ):
                        self.module.main()

                self.assertEqual(delete_deploy_key_mock.call_count, 1)
                self.assertEqual(create_deploy_key_mock.call_count, 1)
                self.assertEqual(exec_info.exception.args[0]["changed"], True)

    @patch.object(BitbucketHelper, "fetch_access_token", return_value="token")
    @patch.object(
        bitbucket_access_key,
        "get_existing_deploy_key",
        return_value={
            "id": 123,
            "label": "mykey",
            "created_on": "2019-03-23T10:15:21.517377+00:00",
            "key": "new public key",
            "type": "deploy_key",
            "comment": "",
            "last_used": None,
            "repository": {
                "links": {
                    "self": {"href": "https://api.bitbucket.org/2.0/repositories/mleu/test"},
                    "html": {"href": "https://bitbucket.org/mleu/test"},
                    "avatar": {"href": "..."},
                },
                "type": "repository",
                "name": "test",
                "full_name": "mleu/test",
                "uuid": "{85d08b4e-571d-44e9-a507-fa476535aa98}",
            },
            "links": {"self": {"href": "https://api.bitbucket.org/2.0/repositories/mleu/test/deploy-keys/123"}},
        },
    )
    def test_dont_update_same_value(self, *args):
        with patch.object(self.module, "delete_deploy_key") as delete_deploy_key_mock:
            with patch.object(self.module, "create_deploy_key") as create_deploy_key_mock:
                with self.assertRaises(AnsibleExitJson) as exec_info:
                    with set_module_args(
                        {
                            "client_id": "ABC",
                            "client_secret": "XXX",
                            "workspace": "name",
                            "repository": "repo",
                            "key": "new public key",
                            "label": "mykey",
                            "state": "present",
                        }
                    ):
                        self.module.main()

                self.assertEqual(delete_deploy_key_mock.call_count, 0)
                self.assertEqual(create_deploy_key_mock.call_count, 0)
                self.assertEqual(exec_info.exception.args[0]["changed"], False)

    @patch.object(BitbucketHelper, "fetch_access_token", return_value="token")
    @patch.object(
        bitbucket_access_key,
        "get_existing_deploy_key",
        return_value={
            "id": 123,
            "label": "mykey",
            "created_on": "2019-03-23T10:15:21.517377+00:00",
            "key": "ssh-rsa AAAAB3NzaC1yc2EAAAADA...AdkTg7HGqL3rlaDrEcWfL7Lu6TnhBdq5",
            "type": "deploy_key",
            "comment": "",
            "last_used": None,
            "repository": {
                "links": {
                    "self": {"href": "https://api.bitbucket.org/2.0/repositories/mleu/test"},
                    "html": {"href": "https://bitbucket.org/mleu/test"},
                    "avatar": {"href": "..."},
                },
                "type": "repository",
                "name": "test",
                "full_name": "mleu/test",
                "uuid": "{85d08b4e-571d-44e9-a507-fa476535aa98}",
            },
            "links": {"self": {"href": "https://api.bitbucket.org/2.0/repositories/mleu/test/deploy-keys/123"}},
        },
    )
    def test_update_deploy_key_check_mode(self, *args):
        with patch.object(self.module, "delete_deploy_key") as delete_deploy_key_mock:
            with patch.object(self.module, "create_deploy_key") as create_deploy_key_mock:
                with self.assertRaises(AnsibleExitJson) as exec_info:
                    with set_module_args(
                        {
                            "client_id": "ABC",
                            "client_secret": "XXX",
                            "workspace": "name",
                            "repository": "repo",
                            "key": "new public key",
                            "label": "mykey",
                            "state": "present",
                            "_ansible_check_mode": True,
                        }
                    ):
                        self.module.main()

                self.assertEqual(delete_deploy_key_mock.call_count, 0)
                self.assertEqual(create_deploy_key_mock.call_count, 0)
                self.assertEqual(exec_info.exception.args[0]["changed"], True)

    @patch.object(BitbucketHelper, "fetch_access_token", return_value="token")
    @patch.object(
        bitbucket_access_key,
        "get_existing_deploy_key",
        return_value={
            "id": 123,
            "label": "mykey",
            "created_on": "2019-03-23T10:15:21.517377+00:00",
            "key": "ssh-rsa AAAAB3NzaC1yc2EAAAADA...AdkTg7HGqL3rlaDrEcWfL7Lu6TnhBdq5",
            "type": "deploy_key",
            "comment": "",
            "last_used": None,
            "repository": {
                "links": {
                    "self": {"href": "https://api.bitbucket.org/2.0/repositories/mleu/test"},
                    "html": {"href": "https://bitbucket.org/mleu/test"},
                    "avatar": {"href": "..."},
                },
                "type": "repository",
                "name": "test",
                "full_name": "mleu/test",
                "uuid": "{85d08b4e-571d-44e9-a507-fa476535aa98}",
            },
            "links": {"self": {"href": "https://api.bitbucket.org/2.0/repositories/mleu/test/deploy-keys/123"}},
        },
    )
    def test_delete_deploy_key(self, *args):
        with patch.object(self.module, "delete_deploy_key") as delete_deploy_key_mock:
            with self.assertRaises(AnsibleExitJson) as exec_info:
                with set_module_args(
                    {
                        "client_id": "ABC",
                        "client_secret": "XXX",
                        "workspace": "name",
                        "repository": "repo",
                        "label": "mykey",
                        "state": "absent",
                    }
                ):
                    self.module.main()

            self.assertEqual(delete_deploy_key_mock.call_count, 1)
            self.assertEqual(exec_info.exception.args[0]["changed"], True)

    @patch.object(BitbucketHelper, "fetch_access_token", return_value="token")
    @patch.object(bitbucket_access_key, "get_existing_deploy_key", return_value=None)
    def test_delete_absent_deploy_key(self, *args):
        with patch.object(self.module, "delete_deploy_key") as delete_deploy_key_mock:
            with self.assertRaises(AnsibleExitJson) as exec_info:
                with set_module_args(
                    {
                        "client_id": "ABC",
                        "client_secret": "XXX",
                        "workspace": "name",
                        "repository": "repo",
                        "label": "mykey",
                        "state": "absent",
                    }
                ):
                    self.module.main()

            self.assertEqual(delete_deploy_key_mock.call_count, 0)
            self.assertEqual(exec_info.exception.args[0]["changed"], False)

    @patch.object(BitbucketHelper, "fetch_access_token", return_value="token")
    @patch.object(
        bitbucket_access_key,
        "get_existing_deploy_key",
        return_value={
            "id": 123,
            "label": "mykey",
            "created_on": "2019-03-23T10:15:21.517377+00:00",
            "key": "ssh-rsa AAAAB3NzaC1yc2EAAAADA...AdkTg7HGqL3rlaDrEcWfL7Lu6TnhBdq5",
            "type": "deploy_key",
            "comment": "",
            "last_used": None,
            "repository": {
                "links": {
                    "self": {"href": "https://api.bitbucket.org/2.0/repositories/mleu/test"},
                    "html": {"href": "https://bitbucket.org/mleu/test"},
                    "avatar": {"href": "..."},
                },
                "type": "repository",
                "name": "test",
                "full_name": "mleu/test",
                "uuid": "{85d08b4e-571d-44e9-a507-fa476535aa98}",
            },
            "links": {"self": {"href": "https://api.bitbucket.org/2.0/repositories/mleu/test/deploy-keys/123"}},
        },
    )
    def test_delete_deploy_key_check_mode(self, *args):
        with patch.object(self.module, "delete_deploy_key") as delete_deploy_key_mock:
            with self.assertRaises(AnsibleExitJson) as exec_info:
                with set_module_args(
                    {
                        "client_id": "ABC",
                        "client_secret": "XXX",
                        "workspace": "name",
                        "repository": "repo",
                        "label": "mykey",
                        "state": "absent",
                        "_ansible_check_mode": True,
                    }
                ):
                    self.module.main()

            self.assertEqual(delete_deploy_key_mock.call_count, 0)
            self.assertEqual(exec_info.exception.args[0]["changed"], True)


if __name__ == "__main__":
    unittest.main()
